STORM: Stepwise Token Optimization with Reward-Guided Beam Search
STORM:奖励引导束搜索的逐步 token 优化查询扩展
🧭 查询改写与词汇扩展 · 奖励引导束搜索 ★ 深度解读
MILA – Quebec AI Institute & ILLS · Université Paris-Saclay & CentraleSupélec & CNRS · Air Liquide · Sorbonne Université & ISIR & CNRS · Sapienza, University of Rome
将检索奖励转化为 token 级信号,通过奖励引导束搜索训练 LLM 生成高效关键词,在不改变索引下大幅提升 BM25
📌 工作参考价值(4 点)
- 奖励信号粒度下放:将检索质量指标(如 nDCG)转化为 token 级反馈,在生成过程中实时剪枝,解决序列级奖励延迟与稀疏问题。这一思路可直接用于电商搜索/推荐中的 query 改写、标题优化、对话式 Agent 的关键词生成,用业务指标(CTR、转化率)替换 nDCG 即可训练。
- 低成本高效重写:STORM 训练的 LLM 输出短关键词串,推理延迟接近 BM25,适合在线实时场景。电商搜索可部署为轻量级改写模块,不增加检索延迟,却显著提升召回与相关性。
- 自监督训练摆脱人工标注:无需人工改写对,只需相关性判断或近似信号(如点击日志、伪标签)。可借鉴其利用交叉编码器伪标签蒸馏到生成式改写器的方法,在标注稀缺的电商场景快速冷启动。
- 多语言零样本迁移:仅用英文 MS-MARCO 训练,即可迁移到多语言 BM25 索引,效果超越专用多语言稠密检索器。对跨国电商平台,可简化多语 query 改写部署,无需额外训练。